package cos_api

import (
	"context"
	"fmt"
	"github.com/tencentyun/cos-go-sdk-v5"
	"io"
	"log"
	"net/http"
	"net/url"
	"partyus_app/config"
)

func getClient() (*cos.Client, string) {
	// 存储桶名称，由 bucketname-appid 组成，appid 必须填入，可以在 COS 控制台查看存储桶名称。 https://console.cloud.tencent.com/cos5/bucket
	// 替换为用户的 region，存储桶 region 可以在 COS 控制台“存储桶概览”查看 https://console.cloud.tencent.com/ ，关于地域的详情见 https://cloud.tencent.com/document/product/436/6224 。
	bucketAID := config.GlobalConfig.Cos.BucketAID
	args := fmt.Sprintf("https://%s.cos.ap-guangzhou.myqcloud.com", bucketAID)
	u, _ := url.Parse(args)
	b := &cos.BaseURL{BucketURL: u}
	return cos.NewClient(b, &http.Client{
		Transport: &cos.AuthorizationTransport{
			// 通过环境变量获取密钥
			// 环境变量 SECRETID 表示用户的 SecretId，登录访问管理控制台查看密钥，https://console.cloud.tencent.com/cam/capi
			SecretID: config.GlobalConfig.Cos.SecretID, // 用户的 SecretId，建议使用子账号密钥，授权遵循最小权限指引，降低使用风险。子账号密钥获取可参见 https://cloud.tencent.com/document/product/598/37140
			// 环境变量 SECRETKEY 表示用户的 SecretKey，登录访问管理控制台查看密钥，https://console.cloud.tencent.com/cam/capi
			SecretKey: config.GlobalConfig.Cos.SecretKey, // 用户的 SecretKey，建议使用子账号密钥，授权遵循最小权限指引，降低使用风险。子账号密钥获取可参见 https://cloud.tencent.com/document/product/598/37140
		},
	}), args
}

// UploadFileToCOS 上传文件到腾讯云 COS
func UploadFileToCOS(localPath, objectKey string) (string, error) {
	client, baseURL := getClient()

	// 上传文件到 COS
	_, err := client.Object.PutFromFile(context.Background(), objectKey, localPath, nil)
	if err != nil {
		return "", err
	}

	// 返回拼接好的 URL
	url := fmt.Sprintf("%s/%s", baseURL, objectKey)
	return url, nil
}

func UploadReaderToCOS(reader io.Reader, objectKey string) (string, error) {
	client, baseURL := getClient()

	// 直接上传 Reader 数据流
	_, err := client.Object.Put(context.Background(), objectKey, reader, nil)
	if err != nil {
		log.Print(err)
		return "", err
	}

	url := fmt.Sprintf("%s/%s", baseURL, objectKey)
	return url, nil
}
